/***
  This file is part of PulseAudio.

  Copyright 2004-2006 Lennart Poettering
  Copyright 2006 Pierre Ossman <ossman@cendio.se> for Cendio AB

  PulseAudio is free software; you can redistribute it and/or modify
  it under the terms of the GNU Lesser General Public License as published
  by the Free Software Foundation; either version 2.1 of the License,
  or (at your option) any later version.

  PulseAudio is distributed in the hope that it will be useful, but
  WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  General Public License for more details.

  You should have received a copy of the GNU Lesser General Public License
  along with PulseAudio; if not, write to the Free Software
  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  USA.
***/

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif


#include <pulsecore/macro.h>
#include <pulsecore/g711.h>
#include <pulsecore/endianmacros.h>

#include "sample-util.h"

#undef DISABLE_FIXED_VOLUME

static bool fixed_volume(int32_t *volumes, unsigned channels)
{
	unsigned channel;
    unsigned fast_vol;
    int32_t same_vol = volumes[0];

#ifdef DISABLE_FIXED_VOLUME
	return 0;
#endif

    for (channel = 0; channel < channels; channel++) {
	    if (volumes[channel] != same_vol)
		    break;
    }

    fast_vol = channel < channels? 0 : 1;

	if (channels == 1)
		fast_vol = 0;

	return fast_vol;
}

static void pa_volume_u8_c(uint8_t *samples, int32_t *volumes, unsigned channels, unsigned length) {
    unsigned channel;

    int32_t same_vol = volumes[0];
    unsigned fast_vol;
    fast_vol = fixed_volume(volumes, channels);

    if (fast_vol) {
	int32_t t, hi, lo;
	int32_t ht, lt;
	int32_t s0, s1, s2, s3;
	uint8_t t0, t1, t2, t3;

	hi = same_vol >> 16;
	lo = same_vol & 0xffff;
	while (length) {
		t = *((int32_t *)samples);

		s0 = t & 0xFF;
		s1 = (t >> 8) & 0xFF;
		s2 = (t >> 16) & 0xFF;
		s3 = (t >> 24) & 0xFF;

	        t = s0 - 0x80;
		t = ((t * lo) >> 16) + (t * hi);
		t = PA_CLAMP_UNLIKELY(t, -0x80, 0x7F);
		t0 = (uint8_t) (t + 0x80);

		t = s1 - 0x80;
		t = ((t * lo) >> 16) + (t * hi);
		t = PA_CLAMP_UNLIKELY(t, -0x80, 0x7F);
		t1 = (uint8_t)(t + 0x80);
	    t = s2 - 0x80;
		t = ((t * lo) >> 16) + (t * hi);
	    t = PA_CLAMP_UNLIKELY(t, -0x80, 0x7F);
		t2 = (uint8_t)(t + 0x80);
        t = s3 - 0x80;
        t = ((t * lo) >> 16) + (t * hi);
	    t = PA_CLAMP_UNLIKELY(t, -0x80, 0x7F);
		t3 = (uint8_t)(t + 0x80);
		*((int32_t *)samples) = (t3 << 24) | (t2 << 16) | (t1 << 8) | t0;

		samples += 4;
		length -= 4;
	}

	return;
    }

    for (channel = 0; length; length--) {
        int32_t t, hi, lo;

        hi = volumes[channel] >> 16;
        lo = volumes[channel] & 0xFFFF;

        t = (int32_t) *samples - 0x80;
        t = ((t * lo) >> 16) + (t * hi);
        t = PA_CLAMP_UNLIKELY(t, -0x80, 0x7F);
        *samples++ = (uint8_t) (t + 0x80);

        if (PA_UNLIKELY(++channel >= channels))
            channel = 0;
    }
}

static void pa_volume_alaw_c(uint8_t *samples, int32_t *volumes, unsigned channels, unsigned length) {
    unsigned channel;
    int32_t same_vol = volumes[0];
    unsigned fast_vol;

	fast_vol = fixed_volume(volumes, channels);

    if (fast_vol) {
		int32_t t, hi, lo;
		int32_t ht, lt;
		int32_t s0, s1, s2, s3;
		uint8_t t0, t1, t2, t3;

		hi = same_vol >> 16;
		lo = same_vol & 0xffff;

	while (length) {
		t = *((int32_t *)samples);

		s0 = t & 0xFF;
		s1 = (t >> 8) & 0xFF;
		s2 = (t >> 16) & 0xFF;
		s3 = (t >> 24) & 0xFF;

		/* for byte 0 */
        t = (int32_t) st_alaw2linear16(s0);
        t = ((t * lo) >> 16) + (t * hi);
        t = PA_CLAMP_UNLIKELY(t, -0x8000, 0x7FFF);
        t0 = (uint8_t) st_13linear2alaw((int16_t) t >> 3);

        t = (int32_t) st_alaw2linear16(s1);
        t = ((t * lo) >> 16) + (t * hi);
        t = PA_CLAMP_UNLIKELY(t, -0x8000, 0x7FFF);
        t1 = (uint8_t) st_13linear2alaw((int16_t) t >> 3);

        t = (int32_t) st_alaw2linear16(s2);
        t = ((t * lo) >> 16) + (t * hi);
        t = PA_CLAMP_UNLIKELY(t, -0x8000, 0x7FFF);
        t2 = (uint8_t) st_13linear2alaw((int16_t) t >> 3);
        t = (int32_t) st_alaw2linear16(s3);
        t = ((t * lo) >> 16) + (t * hi);
        t = PA_CLAMP_UNLIKELY(t, -0x8000, 0x7FFF);
        t3 = (uint8_t) st_13linear2alaw((int16_t) t >> 3);
		*((int32_t *)samples) = (t3 << 24) | (t2 << 16) | (t1 << 8) | t0;

		samples += 4;
		length -= 4;
	}

	return;
    }

    for (channel = 0; length; length--) {
        int32_t t, hi, lo;

        hi = volumes[channel] >> 16;
        lo = volumes[channel] & 0xFFFF;

        t = (int32_t) st_alaw2linear16(*samples);
        t = ((t * lo) >> 16) + (t * hi);
        t = PA_CLAMP_UNLIKELY(t, -0x8000, 0x7FFF);
        *samples++ = (uint8_t) st_13linear2alaw((int16_t) t >> 3);

        if (PA_UNLIKELY(++channel >= channels))
            channel = 0;
    }
}

static void pa_volume_ulaw_c(uint8_t *samples, int32_t *volumes, unsigned channels, unsigned length) {
    unsigned channel;
    int32_t same_vol = volumes[0];
    unsigned fast_vol;

	fast_vol = fixed_volume(volumes, channels);

    if (fast_vol) {
		int32_t t, hi, lo;
		int32_t ht, lt;
		int32_t s0, s1, s2, s3;
		uint8_t t0, t1, t2, t3;

		hi = same_vol >> 16;
		lo = same_vol & 0xffff;

	while (length) {
		t = *((int32_t *)samples);
		s0 = t & 0xFF;
		s1 = (t >> 8) & 0xFF;
		s2 = (t >> 16) & 0xFF;
		s3 = (t >> 24) & 0xFF;

		/* for byte 0 */
        t = (int32_t) st_ulaw2linear16(s0);
        t = ((t * lo) >> 16) + (t * hi);
        t = PA_CLAMP_UNLIKELY(t, -0x8000, 0x7FFF);
        t0 = (uint8_t) st_14linear2ulaw((int16_t) t >> 2);

        t = (int32_t) st_ulaw2linear16(s1);
        t = ((t * lo) >> 16) + (t * hi);
        t = PA_CLAMP_UNLIKELY(t, -0x8000, 0x7FFF);
        t1 = (uint8_t) st_14linear2ulaw((int16_t) t >> 2);

        t = (int32_t) st_ulaw2linear16(s2);
        t = ((t * lo) >> 16) + (t * hi);
        t = PA_CLAMP_UNLIKELY(t, -0x8000, 0x7FFF);
        t2 = (uint8_t) st_14linear2ulaw((int16_t) t >> 2);
        t = (int32_t) st_ulaw2linear16(s3);
        t = ((t * lo) >> 16) + (t * hi);
        t = PA_CLAMP_UNLIKELY(t, -0x8000, 0x7FFF);
        t3 = (uint8_t) st_14linear2ulaw((int16_t) t >> 2);
		*((int32_t *)samples) = (t3 << 24) | (t2 << 16) | (t1 << 8) | t0;

		samples += 4;
		length -= 4;
	}

	return;
    }

    for (channel = 0; length; length--) {
        int32_t t, hi, lo;

        hi = volumes[channel] >> 16;
        lo = volumes[channel] & 0xFFFF;

        t = (int32_t) st_ulaw2linear16(*samples);
        t = ((t * lo) >> 16) + (t * hi);
        t = PA_CLAMP_UNLIKELY(t, -0x8000, 0x7FFF);
        *samples++ = (uint8_t) st_14linear2ulaw((int16_t) t >> 2);

        if (PA_UNLIKELY(++channel >= channels))
            channel = 0;
    }
}

static void pa_volume_s16ne_c(int16_t *samples, int32_t *volumes, unsigned channels, unsigned length) {
    unsigned channel;
    int32_t same_vol = volumes[0];
    unsigned fast_vol;

    length /= sizeof(int16_t);

	fast_vol = fixed_volume(volumes, channels);

    if (fast_vol) {
		int32_t t, hi, lo;
		int32_t ht, lt;

		hi = same_vol >> 16;
		lo = same_vol & 0xffff;

	while (length) {
		t = *((int32_t *)samples);	
		ht = t >> 16; 
		lt = t & 0xffff;

        	ht = ((ht * lo) >> 16) + (ht * hi);
	        ht = PA_CLAMP_UNLIKELY(ht, -0x8000, 0x7FFF);

        	lt = ((lt * lo) >> 16) + (lt * hi);
	        lt = PA_CLAMP_UNLIKELY(lt, -0x8000, 0x7FFF);
		
		*((int32_t *)samples) = ht<<16|lt;

		samples += 2;
		length -= 2;
	}

	return;
    }
    for (channel = 0; length; length--) {
        int32_t t, hi, lo;

        /* Multiplying the 32bit volume factor with the 16bit
         * sample might result in an 48bit value. We want to
         * do without 64 bit integers and hence do the
         * multiplication independently for the HI and LO part
         * of the volume. */

        hi = volumes[channel] >> 16;
        lo = volumes[channel] & 0xFFFF;

        t = (int32_t)(*samples);
        t = ((t * lo) >> 16) + (t * hi);
        t = PA_CLAMP_UNLIKELY(t, -0x8000, 0x7FFF);
        *samples++ = (int16_t) t;

        if (PA_UNLIKELY(++channel >= channels))
            channel = 0;
    }
}

static void pa_volume_s16re_c(int16_t *samples, int32_t *volumes, unsigned channels, unsigned length) {
    unsigned channel;
    int32_t same_vol = volumes[0];
    unsigned fast_vol;

    length /= sizeof(int16_t);

	fast_vol = fixed_volume(volumes, channels);

    if (fast_vol) {
		int32_t t, hi, lo;
		int32_t ht, lt;

		hi = same_vol >> 16;
		lo = same_vol & 0xffff;

	while (length) {
		t = *((int32_t *)samples);	
		ht = (int32_t)PA_INT16_SWAP((int16_t)(t >> 16)); 
		lt = (int32_t)PA_INT16_SWAP((int16_t)(t & 0xffff));

        	ht = ((ht * lo) >> 16) + (ht * hi);
	        ht = PA_CLAMP_UNLIKELY(ht, -0x8000, 0x7FFF);

        	lt = ((lt * lo) >> 16) + (lt * hi);
	        lt = PA_CLAMP_UNLIKELY(lt, -0x8000, 0x7FFF);
		
		*((int32_t *)samples) = PA_INT16_SWAP((int16_t) ht<<16)|PA_INT16_SWAP((int16_t) lt);

		samples += 2;
		length -= 2;
	}

	return;
    }

    for (channel = 0; length; length--) {
        int32_t t, hi, lo;

        hi = volumes[channel] >> 16;
        lo = volumes[channel] & 0xFFFF;

        t = (int32_t) PA_INT16_SWAP(*samples);
        t = ((t * lo) >> 16) + (t * hi);
        t = PA_CLAMP_UNLIKELY(t, -0x8000, 0x7FFF);
        *samples++ = PA_INT16_SWAP((int16_t) t);

        if (PA_UNLIKELY(++channel >= channels))
            channel = 0;
    }
}

static void pa_volume_float32ne_c(float *samples, float *volumes, unsigned channels, unsigned length) {
    unsigned channel;

    length /= sizeof(float);

    for (channel = 0; length; length--) {
        *samples++ *= volumes[channel];

        if (PA_UNLIKELY(++channel >= channels))
            channel = 0;
    }
}

static void pa_volume_float32re_c(float *samples, float *volumes, unsigned channels, unsigned length) {
    unsigned channel;

    length /= sizeof(float);

    for (channel = 0; length; length--) {
        float t;

        t = PA_FLOAT32_SWAP(*samples);
        t *= volumes[channel];
        *samples++ = PA_FLOAT32_SWAP(t);

        if (PA_UNLIKELY(++channel >= channels))
            channel = 0;
    }
}

static void pa_volume_s32ne_c(int32_t *samples, int32_t *volumes, unsigned channels, unsigned length) {
    unsigned channel;

    length /= sizeof(int32_t);

    for (channel = 0; length; length--) {
        int64_t t;

        t = (int64_t)(*samples);
        t = (t * volumes[channel]) >> 16;
        t = PA_CLAMP_UNLIKELY(t, -0x80000000LL, 0x7FFFFFFFLL);
        *samples++ = (int32_t) t;

        if (PA_UNLIKELY(++channel >= channels))
            channel = 0;
    }
}

static void pa_volume_s32re_c(int32_t *samples, int32_t *volumes, unsigned channels, unsigned length) {
    unsigned channel;

    length /= sizeof(int32_t);

    for (channel = 0; length; length--) {
        int64_t t;

        t = (int64_t) PA_INT32_SWAP(*samples);
        t = (t * volumes[channel]) >> 16;
        t = PA_CLAMP_UNLIKELY(t, -0x80000000LL, 0x7FFFFFFFLL);
        *samples++ = PA_INT32_SWAP((int32_t) t);

        if (PA_UNLIKELY(++channel >= channels))
            channel = 0;
    }
}

static void pa_volume_s24ne_c(uint8_t *samples, int32_t *volumes, unsigned channels, unsigned length) {
    unsigned channel;
    uint8_t *e;

    e = samples + length;

    for (channel = 0; samples < e; samples += 3) {
        int64_t t;

        t = (int64_t)((int32_t) (PA_READ24NE(samples) << 8));
        t = (t * volumes[channel]) >> 16;
        t = PA_CLAMP_UNLIKELY(t, -0x80000000LL, 0x7FFFFFFFLL);
        PA_WRITE24NE(samples, ((uint32_t) (int32_t) t) >> 8);

        if (PA_UNLIKELY(++channel >= channels))
            channel = 0;
    }
}

static void pa_volume_s24re_c(uint8_t *samples, int32_t *volumes, unsigned channels, unsigned length) {
    unsigned channel;
    uint8_t *e;

    e = samples + length;

    for (channel = 0; samples < e; samples += 3) {
        int64_t t;

        t = (int64_t)((int32_t) (PA_READ24RE(samples) << 8));
        t = (t * volumes[channel]) >> 16;
        t = PA_CLAMP_UNLIKELY(t, -0x80000000LL, 0x7FFFFFFFLL);
        PA_WRITE24RE(samples, ((uint32_t) (int32_t) t) >> 8);

        if (PA_UNLIKELY(++channel >= channels))
            channel = 0;
    }
}

static void pa_volume_s24_32ne_c(uint32_t *samples, int32_t *volumes, unsigned channels, unsigned length) {
    unsigned channel;

    length /= sizeof(uint32_t);

    for (channel = 0; length; length--) {
        int64_t t;

        t = (int64_t) ((int32_t) (*samples << 8));
        t = (t * volumes[channel]) >> 16;
        t = PA_CLAMP_UNLIKELY(t, -0x80000000LL, 0x7FFFFFFFLL);
        *samples++ = ((uint32_t) ((int32_t) t)) >> 8;

        if (PA_UNLIKELY(++channel >= channels))
            channel = 0;
    }
}

static void pa_volume_s24_32re_c(uint32_t *samples, int32_t *volumes, unsigned channels, unsigned length) {
    unsigned channel;

    length /= sizeof(uint32_t);

    for (channel = 0; length; length--) {
        int64_t t;

        t = (int64_t) ((int32_t) (PA_UINT32_SWAP(*samples) << 8));
        t = (t * volumes[channel]) >> 16;
        t = PA_CLAMP_UNLIKELY(t, -0x80000000LL, 0x7FFFFFFFLL);
        *samples++ = PA_UINT32_SWAP(((uint32_t) ((int32_t) t)) >> 8);

        if (PA_UNLIKELY(++channel >= channels))
            channel = 0;
    }
}

static pa_do_volume_func_t do_volume_table[] = {
    [PA_SAMPLE_U8]        = (pa_do_volume_func_t) pa_volume_u8_c,
    [PA_SAMPLE_ALAW]      = (pa_do_volume_func_t) pa_volume_alaw_c,
    [PA_SAMPLE_ULAW]      = (pa_do_volume_func_t) pa_volume_ulaw_c,
    [PA_SAMPLE_S16NE]     = (pa_do_volume_func_t) pa_volume_s16ne_c,
    [PA_SAMPLE_S16RE]     = (pa_do_volume_func_t) pa_volume_s16re_c,
    [PA_SAMPLE_FLOAT32NE] = (pa_do_volume_func_t) pa_volume_float32ne_c,
    [PA_SAMPLE_FLOAT32RE] = (pa_do_volume_func_t) pa_volume_float32re_c,
    [PA_SAMPLE_S32NE]     = (pa_do_volume_func_t) pa_volume_s32ne_c,
    [PA_SAMPLE_S32RE]     = (pa_do_volume_func_t) pa_volume_s32re_c,
    [PA_SAMPLE_S24NE]     = (pa_do_volume_func_t) pa_volume_s24ne_c,
    [PA_SAMPLE_S24RE]     = (pa_do_volume_func_t) pa_volume_s24re_c,
    [PA_SAMPLE_S24_32NE]  = (pa_do_volume_func_t) pa_volume_s24_32ne_c,
    [PA_SAMPLE_S24_32RE]  = (pa_do_volume_func_t) pa_volume_s24_32re_c
};

pa_do_volume_func_t pa_get_volume_func(pa_sample_format_t f) {
    pa_assert(f >= 0);
    pa_assert(f < PA_SAMPLE_MAX);

    return do_volume_table[f];
}

void pa_set_volume_func(pa_sample_format_t f, pa_do_volume_func_t func) {
    pa_assert(f >= 0);
    pa_assert(f < PA_SAMPLE_MAX);

    do_volume_table[f] = func;
}
